import pandas as pd
import os

def mergeData():
    tran_data_dir = './train/'  # 训练数据文件处理跟目录
    all_train_data = pd.DataFrame() # 初始化一个空的 DataFrame 以存储所有训练数据

    count = 1;
    # 遍历指定目录下的所有 CSV 文件
    for csv_file in os.listdir(tran_data_dir):
        # 追加到 DataFrame 中
        if csv_file.endswith('.csv'):
            train_data = pd.read_csv(tran_data_dir + csv_file)
            all_train_data = pd.concat([all_train_data, train_data], ignore_index=True)
        count += 1
        if count % 100 == 0:
            print(f"正在处理第  {count}  个文件")
    # for i in range(len(all_train_data)):
    #     print(all_train_data.iloc[i,:])

    # 保存所有的训练数据到一个csv文件中
    all_train_data.to_csv('all_train_data.csv', index=False)

    return all_train_data
